import numpy as np
import pandas as pd

df = pd.DataFrame(
    {
        'name': ['Tom', 'Nick', 'lily', 'Zhangsan', 'Lisi'],
        'age': [18, 20, 20, 21, 22]
    }
)

# 有条件查询
list1 = [1, 2, 3, 4, 5, 6]
# print(list1 > 3)
# list的比较会使用一整个list进行比较，上述操作是使用list和int进行比较，类型不一致
list1 = np.array(list1).reshape(2, 3)
# 因为数组可以实现遍历操作，通过对数组内每一个成员都与3进行比较，返回布尔值
# print(list1 > 3)  # 数组的比较有广播的功能

# DataFrame操作类似
# 返回了一个bool类型的series
# print(df['age'] > 20)
# 如果需要保存查询到的内容，通过查询，将布尔值True的数值保存，False的数值移除
# 返回的就是搜索完成后的DataFrame
df = df[df['age'] > 20]
print(df)